from utils.registry import Registry
from monai.metrics import DiceMetric

# 评估指标注册器
METRICS = Registry('metric')

@METRICS.register()
class DiceMetricWrapper(DiceMetric):
    def __init__(self, **kwargs):
        super().__init__(**kwargs)